mpirun -n 8 python train.py --config ./src/configs/mlp.yaml